import React from 'react';
import { BasicLayoutProps, Settings as LayoutSettings } from '@ant-design/pro-layout';

import { notification, message } from 'antd';
import { history, RequestConfig } from 'umi';
import { queryCurrent } from './services/user';
import MenuCounter from '@/components/SiderMenu/Counter';
import { pick } from 'lodash';

import defaultSettings from '../config/defaultSettings';
import HeaderRender from '@/components/HeaderRender';
import MenuHeader from '@/components/HeaderRender/MenuHeaderRender';
import menuItemRenderFn from './components/HeaderRender/menuItemRender';
import SiderMenu from '@/components/SiderMenu';

export async function getInitialState(): Promise<{
  currentUser?: API.CurrentUser;
  settings?: LayoutSettings;
  collapsed?: BasicLayoutProps['collapsed'];
}> {
  // 如果是登录页面，不执行
  if (history.location.pathname !== '/user/login') {
    try {
      const res: API.LoginStateType = await queryCurrent();
      const currentUser: API.CurrentUser =
        res?.result === 0 ? pick(res?.data, ['photo', 'name', 'areas', 'useName']) : {};

      return {
        currentUser,
        settings: defaultSettings,
        collapsed: false,
      };
    } catch (error) {
      history.push('/user/login');
    }
  }
  return {
    settings: defaultSettings,
    collapsed: false,
  };
}

export const layout = ({
  initialState,
}: {
  initialState: {
    collapsed?: boolean;
    settings?: LayoutSettings;
    onCollapse?: BasicLayoutProps['onCollapse'];
  };
}): BasicLayoutProps => {
  let contentStyle = initialState?.collapsed ? { contentStyle: { marginLeft: '52px' } } : {};

  return {
    ...contentStyle,
    onCollapse: () => {},
    collapsed: initialState?.collapsed,
    disableContentMargin: false,
    headerRender: () => <HeaderRender />,
    collapsedButtonRender: () => null,
    menuItemRender: menuItemRenderFn,
    menuRender: (props) => {
      return (
        <MenuCounter.Provider>
          <SiderMenu {...props} />
        </MenuCounter.Provider>
      );
    },
    className: initialState?.collapsed ? 'entrance-guard-collapsed' : undefined,
    subMenuItemRender: menuItemRenderFn,
    menuHeaderRender: () => <MenuHeader />,
    ...initialState?.settings,
  };
};

const codeMessage = {
  200: '服务器成功返回请求的数据。',
  201: '新建或修改数据成功。',
  202: '一个请求已经进入后台排队（异步任务）。',
  204: '删除数据成功。',
  400: '发出的请求有错误，服务器没有进行新建或修改数据的操作。',
  401: '用户没有权限（令牌、用户名、密码错误）。',
  403: '用户得到授权，但是访问是被禁止的。',
  404: '发出的请求针对的是不存在的记录，服务器没有进行操作。',
  405: '请求方法不被允许。',
  406: '请求的格式不可得。',
  410: '请求的资源被永久删除，且不会再得到的。',
  422: '当创建一个对象时，发生一个验证错误。',
  500: '服务器发生错误，请检查服务器。',
  502: '网关错误。',
  503: '服务不可用，服务器暂时过载或维护。',
  504: '网关超时。',
};

/**
 * 异常处理程序
 */
const errorHandler = (error: { response: Response }) => {
  const { response } = error;
  if (response && response.status) {
    const errorText = codeMessage[response.status] || response.statusText;
    const { status, url } = response;

    notification.error({
      message: `请求错误 ${status}: ${url}`,
      description: errorText,
    });
  }

  if (!response) {
    notification.error({
      description: '您的网络发生异常，无法连接服务器',
      message: '网络异常',
    });
  }
  throw error;
};
//request 拦截器
const requestInterceptors = (url: string, options: any) => {
  let c_token = localStorage.getItem('token');

  if (c_token) {
    const headers = {
      // 文件导入导出的时候传输方式会有问题
      // 'Content-Type': 'application/json',
      Accept: 'application/json',
      Authorization: c_token,
      ...options.headers,
    };
    return {
      url: url,
      options: { ...options, headers: headers },
    };
  } else {
    const headers = {
      ...options.headers,
      'Content-Type': 'application/json',
      Accept: 'application/json',
    };
    return {
      url: url,
      options: { ...options, headers: headers },
    };
  }
};

export const request: RequestConfig = {
  errorHandler,
  errorConfig: {
    adaptor: (resData) => {
      //请求失败的弹框提示处理
      if (resData && resData.result) {
        message.warning(resData.msg);
      }
      return { ...resData };
    },
  },
  requestInterceptors: [requestInterceptors],
};
